package com.coder;

import com.coder.config.HttpUtils;
import com.coder.entity.CollSat;
import com.coder.service.CollSatService;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

/**
 * @author 王令
 */
@SpringBootTest
public class CollSatReptile {

    @Autowired
    private CollSatService collSatService;

    @Test
    void CollSatReptile() throws IOException {
        List<CollSat> collSats = new ArrayList<>();
        for (int p = 0; p < 124; p++){
            Element bodyElement = HttpUtils.getBodyElement("https://gaokao.chsi.com.cn/zyk/pub/myd/schAppraisalTop.action?start=" + p * 20);
            Elements tr = bodyElement.select(".cnt_table tr");
            for (int i = 1; i < tr.size(); i++) {
                Elements td = tr.get(i).getElementsByTag("td");
                CollSat collSat = new CollSat()
                        .setColleges(td.get(0).text())
                        .setAddress(td.get(1).text())
                        .setComplexSat(td.get(2).text())
                        .setEnvSat(td.get(3).text())
                        .setLifeSat(td.get(4).text());
                collSatService.save(collSat);
//                collSats.add(collSat);
            }
        }

//        System.out.println(collSats.size());
//        for (CollSat collSat : collSats) {
//            System.out.println(collSat);
//        }
    }
}
